/*
 * Copyright 2025 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package androidx.compose.foundation.lazy

import androidx.compose.ui.Modifier
import androidx.compose.ui.layout.BeyondBoundsLayout
import androidx.compose.ui.node.DelegatableNode
import androidx.compose.ui.node.ModifierNodeElement
import androidx.compose.ui.node.findNearestBeyondBoundsLayoutAncestor

internal class ParentCapturingModifierNodeElement(val onParentReady: (BeyondBoundsLayout) -> Unit) :
    ModifierNodeElement<ParentCapturingNode>() {
    override fun create(): ParentCapturingNode = ParentCapturingNode(onParentReady)

    override fun update(node: ParentCapturingNode) {}

    override fun hashCode(): Int = onParentReady.hashCode()

    override fun equals(other: Any?): Boolean =
        other is ParentCapturingModifierNodeElement && other.onParentReady === onParentReady
}

internal class ParentCapturingNode(val onParentReady: (BeyondBoundsLayout) -> Unit) :
    Modifier.Node(), DelegatableNode {
    override fun onAttach() {
        findNearestBeyondBoundsLayoutAncestor()?.let { onParentReady(it) }
    }
}
